package frame

import (
	"stormragetech/dawn/data"

	"gopkg.in/mgo.v2/bson"
)

func init() {
	data.GetConn("test", "sys_subcorp")
}

// SubCorp 下属公司 结构体
type SubCorp struct {
	ID        bson.ObjectId `bson:"_id"`
	SubCorpID string        `bson:"sc_id"`
	Name      string        `bson:"name"`
	CorpID    string        `bson:"corp_id"`
	Locked    bool          `bson:"locked"`
}

// InsSubCorp 新增下属公司
func InsSubCorp(sc SubCorp) error {
	return data.Conn.Insert(sc)
}

// FindBySCID 通过ID查询下属公司
func FindBySCID(id string) (SubCorp, error) {
	sc := SubCorp{}
	err := data.Conn.Find(bson.M{"sc_id": id}).One(&sc)
	return sc, err
}

// ListByCorpID 通过公司ID查询下属公司
func ListByCorpID(id string) ([]SubCorp, error) {
	sc := []SubCorp{}
	err := data.Conn.Find(bson.M{"corp_id": id}).All(&sc)
	return sc, err
}
